# DATA
dataset='Tusimple'
#这里改成绝对路径，相对路径可能会有问题，但是代码上传的时候应该是要用相对路径的
data_root = 'E:/MindSpore/竞赛/代码说明/SHENGTENG_Ultra-Fast-Lane-Detection-master/TUSIMPLEROOT/'

# TRAIN
epoch = 100
batch_size = 8
optimizer = 'Adam'    #['SGD','Adam']
# learning_rate = 0.1
learning_rate = 4e-4
weight_decay = 1e-4
momentum = 0.9

scheduler = 'cos'     #['multi', 'cos']
# steps = [50,75]
gamma  = 0.1
warmup = 'linear'
warmup_iters = 100

# NETWORK
backbone = '18'
griding_num = 100
use_aux = True

# LOSS
sim_loss_w = 1.0
shp_loss_w = 0.0

# EXP
note = ''

log_path = './log/'

# FINETUNE or RESUME MODEL PATH
finetune = None
resume = None

# TEST
test_model = "E:\MindSpore\竞赛\代码说明\SHENGTENG_Ultra-Fast-Lane-Detection-master\ep077.pth"
test_work_dir = "E:/MindSpore/竞赛/代码说明/SHENGTENG_Ultra-Fast-Lane-Detection-master/tmp"

num_lanes = 4